package com.just.fun.arithmetic;

import java.util.Arrays;
import java.util.LinkedList;

/**
 * 异或算法题测试
 */
public class XORTest {
    public static void main(String[] args) {
//        int[] nums1 = new int[]{1, 2, 3, 2, 1};
//        int onlyOne = findOnlyOne(nums1);
//        System.out.println("找出唯一无重复的数：" + onlyOne);

        int[] nums2 = new int[]{1, 2, 3, 2, 1, 4};
        int[] twoDupNum = findTwoDupNum(nums2);
        System.out.println("找出两个无重复的数：" + Arrays.toString(twoDupNum));
    }

    /**
     * 数组中两两重复，但是其中有一个数没重复的，找出数组中唯一无重复的数
     * 如果没有返回-1
     *
     * @param nums
     * @return
     */
    public static int findOnlyOne(int[] nums) {
        if (nums.length <= 2) {
            return -1;
        }
        int res = nums[0];
        for (int i = 1; i < nums.length; i++) {
            res = res ^ nums[i];
        }
        return res;
    }

    /***
     * 数组中两两重复，其中有两个数没重复，找到这两个数
     * 1.先遍历一遍数组，异或得到一个数N。这个树就是不同的两个数异或的结果
     * 2.然后找到N的最低为1的位（假设为J位）
     * 3.再次遍历数组，将按倒数第J位为1和为0，将数组分为两个数组，此时两个数字被分到了两组，每个组再按照找1的方法找就行了。
     * @param nums
     * @return
     *
     */
    public static int[] findTwoDupNum(int[] nums) {
        int n = findOnlyOne(nums);
        //记录右移次数
        int j = 0;
        while ((n & 1) != 1) {
            n = n >> 1;
            j++;
        }
        System.out.println("第一次异或结果：" + n);
        System.out.println("右移位数:" + j);

        int[] result = new int[2];
        //倒数第J位为1的集合
        LinkedList<Integer> int1 = new LinkedList<>();
        //倒数第J位为0的集合
        LinkedList<Integer> int2 = new LinkedList<>();
        for (int i = 0; i < nums.length; i++) {
            if ((1 & nums[i] >> j) == 1) {
                int1.add(nums[i]);
                result[0] = result[0] ^ nums[i];
            } else {
                int2.add(nums[i]);
                result[1] = result[1] ^ nums[i];
            }
        }

        System.out.println("倒数第j位为1的集合：" + int1);
        System.out.println("倒数第j位为0的集合：" + int2);
        return result;

    }
}
